def image_mean_std():
    # 转换图像格式为Tensor
    transform=transforms.Compose([transforms.ToTensor(),])
    # 加载数据集
    my_train = NumberDataset(path="data/captcha/train", type_transform=transform)
    images = []
    # 遍历数据集中的所有图像
    for img, _ in tqdm(my_train):
        images.append(img)
    # 将列表中的所有图像数据组成一个numpy数组
    images = np.stack(images, axis=0)
    # 计算数据集中所有图像的像素平均值（后续正则化的时候需要用到的参数)
    total_mean = np.mean(images, axis=(0, 2, 3))
    # 计算数据集中所有图像的像素标准差（后续正则化的时候需要用到的参数)
    total_std = np.std(images, axis=(0, 2, 3))
    print(total_mean, total_std)
# 执行
image_mean_std()  
